from pydantic import Field
import typing as T
from .ChatBaseModel import ChatBaseModel


class ChatPlusRequestDTO(ChatBaseModel):
    top_n: T.Optional[int] = Field(default=1, description="知识检索答案数量【可选】")

    knowledgeBaseId: T.Optional[str] = Field(
        default=None, description="知识库id【必须在请求头中传递】"
    )
    exclude_image: T.Optional[bool] = Field(
        default=True, description="是否排除图片【可选】"
    )
